python assign function to variable